from langchain_openai import ChatOpenAI
from globle import envs
from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import JsonOutputParser

prompt = '''
需求：
1."今日，央行发布公告宣布降低利率，以刺激经济增长。这一降息举措将影响贷款利率，并在未来几个季度内对金融市场产生影响。"
2."本公司宣布成功收购一家在创新科技领域领先的公司，这一战略性收购将有助于公司拓展技术能力和加速产品研发。"
3."公司资产负债表显示，公司偿债能力强劲，现金流充足，为未来投资和扩张提供了坚实的财务基础。"
4."最新的分析报告指出，可再生能源行业预计将在未来几年经历持续增长，投资者应该关注这一领域的投资机会"
把这些分类，类别范围['新闻报道',  '公司公告',  '财务公告', '分析师报告']
将输出按照json形式输出,字段
- list [类别1，类别2，类别3，类别4]
'''
# 创建模型
model = ChatOpenAI(
    model="deepseek-chat",  # 或具体模型名
    base_url="https://api.deepseek.com",  # DeepSeek API 地址,
    temperature=0.1
)
chain = PromptTemplate.from_template( prompt)|model|JsonOutputParser()
r = chain.invoke({})
print(r['list'])
